#!/bin/bash

function select_network_interface() {
    # 获取到所有的网卡并且将这些网卡作为变量NETWORK_NAME的选项选择后赋值给NETWORK_NAME
    readarray -t NETWORK_OPTIONS < <(ip link show | awk -F: '$0 !~ "lo|vir|docker|br-|veth|wl|^[^0-9]"{print $2}')
    # NETWORK_OPTIONS=("${NETWORK_OPTIONS[@]}")
    PS3="请选择要配置的网络接口："
    select NETWORK_NAME in "${NETWORK_OPTIONS[@]}"; do
        if [[ -n "$NETWORK_NAME" ]]; then
            
            break
        else
            return -1
        fi
    done
    NETWORK_NAME=$(echo "$NETWORK_NAME" | xargs)
    echo "$NETWORK_NAME"
}

function get_current_ip() {
    local NETWORK_NAME="$1"
    # 获取当前主机的ip
    ip=$(ip addr | grep ${NETWORK_NAME} | grep inet | grep -v inet6 | awk '{print $2}' | cut -d/ -f1)
    echo "$ip"
}

function change_ubuntu_mirror() {
    # 获取Ubuntu版本号
    UBUNTU_VERSION=$(lsb_release -rs)

    # 判断Ubuntu版本并替换对应源
    if [[ "$UBUNTU_VERSION" == "22.04" ]]; then
        # 替换 /etc/apt/sources.list 中的源为清华源
        sudo sed -i 's#http://archive.ubuntu.com/ubuntu/#https://mirrors.tuna.tsinghua.edu.cn/ubuntu/#g' /etc/apt/sources.list
        sudo sed -i 's#http://security.ubuntu.com/ubuntu/#https://mirrors.tuna.tsinghua.edu.cn/ubuntu/#g' /etc/apt/sources.list
        echo "已将Ubuntu 22.04 的源替换为清华源。"
    elif [[ "$UBUNTU_VERSION" == "24.04" ]]; then
        # 替换 /etc/apt/sources.list.d/ubuntu.sources 中的源为清华源
        sudo sed -i 's#http://cn.archive.ubuntu.com/ubuntu/#https://mirrors.tuna.tsinghua.edu.cn/ubuntu/#g' /etc/apt/sources.list.d/ubuntu.sources
        echo "已将Ubuntu 24.04 的源替换为清华源。"
    else
        echo "当前系统不是Ubuntu 22.04 或 24.04，不支持此脚本。"
        exit 1
    fi

    # 更新软件包列表
    sudo apt update
    echo "源更换完成并已更新软件包列表。"
}